import java.io.FileReader;

import weka.classifiers.lazy.IBk;
import weka.core.Instance;
import weka.core.Instances;


public class Main {

	public static void main(String[] args) throws Exception {
		FileReader arquivo = new FileReader("campeonato-brasileiro.arff");
		Instances dados = new Instances(arquivo);
		dados.setClassIndex(10);
		Instances baseTeste = dados.testCV(3, 0);
		Instances baseTreino = dados.trainCV(3, 0);
		baseTeste.setClassIndex(10);
		baseTreino.setClassIndex(10);
		IBk knn = new IBk(3);
		knn.buildClassifier(baseTreino);
		
		for (int i = 0; i < baseTeste.numInstances(); i++) {
			Instance teste = baseTeste.instance(i);
			System.out.println("Dado " + (i + 1) + " - " + teste);
			int classe = (int) knn.classifyInstance(teste);
			System.out.println("Knn - " + dados.classAttribute().value(classe));
		}
	}

}
